package com.example.gitstudy.download;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.util.Collections;

/**
 * @author hwj
 * @date 2022/12/27 16:56
 * @描述；文件下载服务
 */
@RestController
@RequestMapping("downLoad")
@Api(tags = "文件下载")
public class DownLoadFile {

    @ApiOperation("以流的形式下载")
    @GetMapping("/downLoadFile")
    public void download(String path,HttpServletResponse response) {
        try {

            String path1 = "F:\\test\\exportTemplate.docx";
            // path是指欲下载的文件的路径。
            File file = new File(path1);
            // 取得文件名。
            String filename = file.getName();
            // 取得文件的后缀名。
            String ext = filename.substring(filename.lastIndexOf(".") + 1).toUpperCase();

            // 以流的形式下载文件。
            InputStream fis = new BufferedInputStream(new FileInputStream(path1));
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();
            // 清空response
            response.reset();
            // 设置response的Header
            response.addHeader("Content-Disposition", "attachment;filename=" + new String(filename.getBytes()));
            response.addHeader("Content-Length", "" + file.length());
            OutputStream toClient = new BufferedOutputStream(response.getOutputStream());
            response.setContentType("application/octet-stream");
            toClient.write(buffer);
            toClient.flush();
            toClient.close();
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }
    @ApiOperation("下载本地文件")
    @GetMapping("/downLoadFile1")
    public void downloadLocal(HttpServletResponse response) throws FileNotFoundException {
        // 下载本地文件
        String fileName = "exportTemplate.docx"; // 文件的默认保存名
        // 读到流中
        InputStream inStream = new FileInputStream("F:\\test\\exportTemplate.docx");// 文件的存放路径
        // 设置输出的格式
        response.reset();
        response.setContentType("bin");
        response.addHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");
        // 循环取出流中的数据
        byte[] b = new byte[100];
        int len;
        try {
            while ((len = inStream.read(b)) > 0)
                response.getOutputStream().write(b, 0, len);
            inStream.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    @ApiOperation("下载网络文件")
    @GetMapping("/downLoadFile2")
    public boolean downloadNet(HttpServletResponse response) throws MalformedURLException {
        // 下载网络文件
        int bytesum = 0;
        int byteread = 0;

        URL url = new URL("https://pics7.baidu.com/feed/574e9258d109b3def1eeecd64f9b158a810a4cee.jpeg"); //网络文件地址

        try {
            URLConnection conn = url.openConnection();
            InputStream inStream = conn.getInputStream();
            FileOutputStream fs = new FileOutputStream("F:/abc.jpeg");//保存的本地地址

            byte[] buffer = new byte[1204];
            int length;
            while ((byteread = inStream.read(buffer)) != -1) {
                bytesum += byteread;
                System.out.println(bytesum);
                fs.write(buffer, 0, byteread);
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return  new File("F:/abc.jpeg").exists();
    }


    @ApiOperation("下载文件--支持在线打开")
    @GetMapping("/downLoadFile3")
    public void downLoad(String filePath, HttpServletResponse response, boolean isOnLine) throws Exception {
        String filePath1 = "F:\\abc.jpeg";
//        String filePath1 = "F:\\test\\exportTemplate.docx";
        File f = new File(filePath1);
        if (!f.exists()) {
            response.sendError(404, "File not found!");
            return;
        }
        BufferedInputStream br = new BufferedInputStream(new FileInputStream(f));
        byte[] buf = new byte[1024];
        int len = 0;
        response.reset(); // 非常重要
        if (isOnLine) { // 在线打开方式
            URL u = new URL("file:///" + filePath1);
            response.setContentType(u.openConnection().getContentType());
            response.setHeader("Content-Disposition", "inline; filename=" + f.getName());
            // 文件名应该编码成UTF-8
        } else { // 纯下载方式
            response.setContentType("application/x-msdownload");
            response.setHeader("Content-Disposition", "attachment; filename=" + f.getName());
        }
        OutputStream out = response.getOutputStream();
        while ((len = br.read(buf)) > 0)
            out.write(buf, 0, len);
        br.close();
        out.close();
    }

    @ApiOperation("文件上传（word）")
    @PostMapping("/importFile")
    public String importExcel(@RequestPart("file") MultipartFile file){
        String originalFilename = file.getOriginalFilename();
        String path = "F:\\test\\图片\\"+originalFilename;
        try {
            System.out.println(originalFilename);
            File file1 = new File(path);
            if(!file1.exists()){
                file1.mkdirs();
            }

            /**
             * 使用默认的transferTo的方法默认上传大小为1m 如果需要上传文件大于1m 可以在配置文件中配置
             */
            file.transferTo(file1);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return path;
    }





}
